import { PrismaClient } from '@prisma/client';

const prisma = new PrismaClient();

async function initProductCategories() {
  try {
    console.log('开始初始化产品分类...');

    const categories = [
      {
        name: 'AI助理',
        description: '智能助手和对话机器人',
        sort: 1
      },
      {
        name: 'AI作画',
        description: 'AI绘画和图像生成工具',
        sort: 2
      },
      {
        name: 'AI视频',
        description: 'AI视频生成和编辑工具',
        sort: 3
      },
      {
        name: 'AI音乐',
        description: 'AI音乐创作和音频处理',
        sort: 4
      },
      {
        name: 'AI编程',
        description: 'AI代码生成和开发工具',
        sort: 5
      },
      {
        name: 'AI写作',
        description: 'AI文本生成和写作助手',
        sort: 6
      },
      {
        name: 'AI设计',
        description: 'AI设计工具和创意助手',
        sort: 7
      },
      {
        name: 'AI办公',
        description: 'AI办公效率工具',
        sort: 8
      },
      {
        name: 'AI教育',
        description: 'AI教育和学习工具',
        sort: 9
      },
      {
        name: 'AI营销',
        description: 'AI营销和推广工具',
        sort: 10
      }
    ];

    for (const category of categories) {
      const existing = await prisma.productCategory.findUnique({
        where: { name: category.name }
      });

      if (!existing) {
        await prisma.productCategory.create({
          data: category
        });
        console.log(`创建分类: ${category.name}`);
      } else {
        console.log(`分类已存在: ${category.name}`);
      }
    }

    console.log('产品分类初始化完成！');
  } catch (error) {
    console.error('初始化产品分类失败:', error);
  } finally {
    await prisma.$disconnect();
  }
}

initProductCategories(); 